Skip to content

Fallback to FP16 from BF16 when MPS is available (for Apple Silicon)#3

Open
cchance27 wants to merge 1 commit intoTTPlanetPig:mainfrom
cchance27:patch-1
Open

Fallback to FP16 from BF16 when MPS is available (for Apple Silicon)#3
cchance27 wants to merge 1 commit intoTTPlanetPig:mainfrom
cchance27:patch-1

Conversation

@cchance27
Copy link
Copy Markdown

  • Added checks for using mps device instead of just cuda.
  • On MPS force models to fp16 instead of AMP Autocast as it's currently not supported on Apple Silicon.

It works on my Macbook M3, using ~19ish gigs similar to BF16, and works well if a bit slow.

Before merging please confirm i didn't break anything on cuda/cpu, as i don't have a pc to test on for that.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant